package com.ats.web.managed.utils;

import javax.annotation.PostConstruct;
import javax.faces.bean.ManagedBean;
import javax.faces.bean.ViewScoped;

import org.primefaces.model.DefaultTreeNode;
import org.primefaces.model.TreeNode;

import com.ats.web.helpers.Document;


@ManagedBean(name = "treeMB")
@ViewScoped
public class TreeMB {

	private TreeNode root;

	@PostConstruct
	public void init() {
		root = createDocuments();
	}

	
	 public TreeNode createDocuments() {
	        TreeNode root = new DefaultTreeNode(new Document("Files", "-", "Folder"), null);
	         
	        TreeNode documents = new DefaultTreeNode(new Document("Documents", "-", "Folder"), root);
	        TreeNode pictures = new DefaultTreeNode(new Document("Pictures", "-", "Folder"), root);
	        TreeNode movies = new DefaultTreeNode(new Document("Movies", "-", "Folder"), root);
	         
	        TreeNode work = new DefaultTreeNode(new Document("Work", "-", "Folder"), documents);
	        TreeNode primefaces = new DefaultTreeNode(new Document("PrimeFaces", "-", "Folder"), documents);
	         
	        //Documents
	        TreeNode expenses = new DefaultTreeNode("document", new Document("Expenses.doc", "30 KB", "Word Document"), work);
	        TreeNode resume = new DefaultTreeNode("document", new Document("Resume.doc", "10 KB", "Word Document"), work);
	        TreeNode refdoc = new DefaultTreeNode("document", new Document("RefDoc.pages", "40 KB", "Pages Document"), primefaces);
	         
	        //Pictures
	        TreeNode barca = new DefaultTreeNode("picture", new Document("barcelona.jpg", "30 KB", "JPEG Image"), pictures);
	        TreeNode primelogo = new DefaultTreeNode("picture", new Document("logo.jpg", "45 KB", "JPEG Image"), pictures);
	        TreeNode optimus = new DefaultTreeNode("picture", new Document("optimusprime.png", "96 KB", "PNG Image"), pictures);
	         
	        //Movies
	        TreeNode pacino = new DefaultTreeNode(new Document("Al Pacino", "-", "Folder"), movies);
	        TreeNode deniro = new DefaultTreeNode(new Document("Robert De Niro", "-", "Folder"), movies);
	         
	        TreeNode scarface = new DefaultTreeNode("mp3", new Document("Scarface", "15 GB", "Movie File"), pacino);
	        TreeNode carlitosWay = new DefaultTreeNode("mp3", new Document("Carlitos' Way", "24 GB", "Movie File"), pacino);
	         
	        TreeNode goodfellas = new DefaultTreeNode("mp3", new Document("Goodfellas", "23 GB", "Movie File"), deniro);
	        TreeNode untouchables = new DefaultTreeNode("mp3", new Document("Untouchables", "17 GB", "Movie File"), deniro);
	         
	        return root;
	    }	
	
	/**
	 * @return the root
	 */
	public TreeNode getRoot() {
		return root;
	}

	/**
	 * @param root
	 *            the root to set
	 */
	public void setRoot(TreeNode root) {
		this.root = root;
	}
}
